package com.chocho.array;

public class SparseArray {
    public static void main(String[] args) {
        //原始二维数组
        //0：没有棋子，1：黑子，2：白子
        int chessArr[][] = new int[11][11];
        chessArr[1][2] = 1;
        chessArr[2][4] = 2;

        System.out.println("原始二维数组:");
        print(chessArr);

        //二维数组转为稀疏数组
        int sparseArr[][] = toSparseArr(chessArr);

        System.out.println("得到的稀疏数组:");
        print(sparseArr);

        //稀疏数组转为二维数组
        int originArr[][] = toArr(sparseArr);
        System.out.println("还原后的二维数组:");
        print(originArr);

    }

    public static void print(int [][]chessArr){
        for (int []line : chessArr){
            for (int data : line){
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }

    /**
     * 二维数组转为稀疏数组
     * @param chessArr 二维数组
     * @return 稀疏数组
     */
    public static int[][] toSparseArr(int [][]chessArr){
        //遍历二维数组，得到总非0个数
        int sum = 0;
        for(int i = 0; i < chessArr.length; i++){
            for(int j = 0; j < chessArr[i].length; j++){
                if(chessArr[i][j] != 0){
                    sum++;
                }
            }
        }
        //创建稀疏数组
        int sparseArr[][] = new int[sum + 1][3];
        //存储原始数组：总行，总列，总非0个数
        sparseArr[0][0] = chessArr.length;
        sparseArr[0][1] = chessArr[0].length;
        sparseArr[0][2] = sum;
        //存储非0数据
        int count = 0; //记录第 count 个非0数据
        for(int i = 0; i < chessArr.length; i++){
            for(int j = 0; j < chessArr[i].length; j++){
                if(chessArr[i][j] != 0){
                    count++;
                    sparseArr[count][0] = i;
                    sparseArr[count][1] = j;
                    sparseArr[count][2] = chessArr[i][j];
                }
            }
        }

        return sparseArr;
    }

    /**
     * 稀疏数组转为二维数组
     * @param sparseArr 稀疏数组
     * @return 二维数组
     */
    public static int[][] toArr(int [][]sparseArr){
        //读取稀疏数组第一行，创建原始数组
        int arr[][] = new int[sparseArr[0][0]][sparseArr[0][1]];

        //读取稀疏数组，将非0数据还原
        for(int i = 1; i < sparseArr.length; i++){
            arr[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
        }

        return arr;
    }
}
